import os

# detect_train = 2485
# detect_val = 277
# pose_train = 1994
# pose_val = 319
# 指定要计算文件数量的文件夹路径
pose_train = r'data\pose4\train\images'


# 使用os.listdir获取文件夹中的所有条目
entries = os.listdir(pose_train)

# 使用os.path.isfile判断每个条目是否是文件，并计数
file_count = sum(1 for entry in entries if os.path.isfile(os.path.join(pose_train, entry)))

print(f"pose_train: {file_count}")
# 指定要计算文件数量的文件夹路径
pose_val = r'data\pose4\val\images'
# detect_train = r'data\detection\atrw_detection_train\trainval\labels\train'


# 使用os.listdir获取文件夹中的所有条目
entries = os.listdir(pose_val)

# 使用os.path.isfile判断每个条目是否是文件，并计数
file_count = sum(1 for entry in entries if os.path.isfile(os.path.join(pose_val, entry)))

print(f"pose_val: {file_count}")
